"""
PyInstaller hook for rembg
"""
from PyInstaller.utils.hooks import collect_data_files, collect_dynamic_libs

# 收集 rembg 的数据文件（包括模型文件等）
datas = collect_data_files('rembg', includes=['**/*.onnx', '**/*.json', '**/*.txt'])

# 收集动态库文件
binaries = collect_dynamic_libs('rembg')

# 隐藏的导入
hiddenimports = [
    'rembg.bg',
    'rembg.session_factory',
    'rembg.sessions',
    'rembg.commands',
    'onnxruntime',
    'PIL',
    'numpy',
    'pooch',
    'requests',
    'scipy',
    'pymatting',
]